package cn.hgy.config;

import cn.hgy.types.annotation.DCCValue;
import cn.hgy.types.common.Constants;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.redisson.api.RBucket;
import org.redisson.api.RTopic;
import org.redisson.api.RedissonClient;
import org.redisson.api.listener.MessageListener;
import org.springframework.aop.framework.AopProxyUtils;
import org.springframework.aop.support.AopUtils;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.lang.reflect.Field;
import java.util.HashMap;
import java.util.Map;

@Slf4j
@Configuration
public class DCCValueBeanFactory implements BeanPostProcessor {
    private static final String BASE_CONFIG_PATH = "group_buy_market_dcc_";

    private final RedissonClient redissonClient;

    private final Map<String, Object> dccObject = new HashMap<>();

    @Bean("dccTopic")
    public RTopic redisTopicListener(RedissonClient redissonClient) {
        RTopic topic = redissonClient.getTopic("group_buy_market_dcc");

        topic.addListener(String.class, new MessageListener<String>() {
            @Override
            public void onMessage(CharSequence charSequence, String s) {
                String[] split = s.split(Constants.SPLIT);

                String attribute = split[0];
                String key = BASE_CONFIG_PATH + attribute;
                String value = split[1];

                RBucket<Object> bucket = redissonClient.getBucket(key);
                boolean exists = bucket.isExists();
                if (!exists) return;;

                bucket.set(value);

                Object objBean = dccObject.get(key);
                if (null == objBean) return;

                Class<?> objBeanClass = objBean.getClass();
                if (AopUtils.isAopProxy(objBeanClass)) {
                    objBeanClass = AopUtils.getTargetClass(objBean);
                }

                try {
                    Field field = objBeanClass.getDeclaredField(attribute);
                    field.setAccessible(true);
                    field.set(objBean, value);
                    field.setAccessible(false);
                }catch (Exception e) {
                    throw new RuntimeException("...");
                }
            }
        });

        return topic;
    }

    public DCCValueBeanFactory(RedissonClient redissonClient) {
        this.redissonClient = redissonClient;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {

        Class<?> targetBeanClass = bean.getClass();
        Object targetBeanObject = bean;

        if (AopUtils.isAopProxy(bean)) {
            targetBeanClass = AopUtils.getTargetClass(bean);
            targetBeanObject = AopProxyUtils.getSingletonTarget(bean);
        }

        Field[] declaredFields = targetBeanClass.getDeclaredFields();
        for (Field field : declaredFields) {
            if (!field.isAnnotationPresent(DCCValue.class)) continue;

            DCCValue dccValue = field.getAnnotation(DCCValue.class);

            String value = dccValue.value();
            if (StringUtils.isBlank(value)) {
                throw new RuntimeException("...");
            }

            String[] split = value.split(":");
            String key = BASE_CONFIG_PATH.concat(split[0]);
            String defaultValue = split.length == 2 ? split[1] : "";

            String setValue = defaultValue;

            if (StringUtils.isBlank(setValue)) {
                throw new RuntimeException("...");
            }

            RBucket<String> bucket = redissonClient.getBucket(key);
            boolean exists = bucket.isExists();
            if (!exists) {
                bucket.set(defaultValue);
            } else {
                setValue = bucket.get();
            }

            field.setAccessible(true);
            try {
                field.set(targetBeanObject, setValue);
            } catch (Exception e) {
                throw new RuntimeException("...");
            }
            field.setAccessible(false);

            dccObject.put(key, targetBeanObject);
        }

        return BeanPostProcessor.super.postProcessAfterInitialization(bean, beanName);
    }
}
